package com.lhb.redis.util;

import org.omg.CORBA.PUBLIC_MEMBER;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.ZSetOperations;
import org.springframework.stereotype.Component;
import org.springframework.util.CollectionUtils;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.TimeUnit;

/**
 * redisTemplate封装
 */
@Component
public class RedisUtil {

    @Autowired
    private RedisTemplate<String,Object> redisTemplate;

    public RedisUtil(RedisTemplate<String,Object> redisTemplate) {
        this.redisTemplate = redisTemplate;
    }

    /**
     * 指定缓存失效时间
     * @param key
     * @param time
     * @return
     */
    public boolean expire(String key,long time){
        try {
            if(time > 0) {
                redisTemplate.expire(key,time, TimeUnit.SECONDS);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 根据key值，获取过期时间
     * @param key 不能为空
     * @return 返回0代表永久有效
     */
    public long getExpire(String key) {
        return redisTemplate.getExpire(key,TimeUnit.SECONDS);
    }

    /**
     * 判断key值是否存在
     * @param key
     * @return
     */
    public boolean hasKey(String key) {
        try {
            return redisTemplate.hasKey(key);
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    @SuppressWarnings("unchecked")
    public void del(String ... key) {
        if(key!=null&&key.length>0){
            if(key.length==1) {
                redisTemplate.delete(key[0]);
            }else {
                redisTemplate.delete(CollectionUtils.arrayToList(key));
            }
        }
    }

    //===================String==========================

    /**
     * 取值
     * @param key
     * @return
     */
    public Object get(String key){
        return key==null?null:redisTemplate.opsForValue().get(key);
    }

    /**
     * 存值
     * @param key
     * @param value
     * @return
     */
    public boolean set(String key,Object value) {
        try {
            redisTemplate.opsForValue().set(key,value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 存值，并设置过期时间
     * @param key
     * @param value
     * @param time 小于0，设置为无限期
     * @return
     */
    public boolean set(String key,Object value,long time) {
        try {
            if(time>0){
                redisTemplate.opsForValue().set(key,value,time,TimeUnit.SECONDS);
            }else {
                set(key,value);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 递增
     * @param key
     * @param delta 要增加几
     * @return
     */
    public long incr(String key,long delta){
        if(delta<0) {
            throw new RuntimeException("递增因子必须大于0");
        }
        return redisTemplate.opsForValue().increment(key,delta);
    }

    /**
     * 递减
     * @param key
     * @param delta 要减少几
     * @return
     */
    public long decr(String key,long delta){
        if(delta<0) {
            throw new RuntimeException("递减因子必须大于0");
        }
        return redisTemplate.opsForValue().decrement(key,-delta);
    }

    //=======================Map===========================================

    /**
     * HashGet
     * @param key 不能为空
     * @param item 不能为空
     * @return
     */
    public Object getHash(String key,String item){
        return redisTemplate.opsForHash().get(key,item);
    }

    /**
     * 获取key对应的所有键值
     * @param key
     * @return
     */
    public Map<Object,Object> getHashMap(String key){
        return redisTemplate.opsForHash().entries(key);
    }

    /**
     * hashSet
     * @param key
     * @param map 对应多个键值
     * @return
     */
    public boolean setHashMap(String key,Map<String,Object> map){
        try {
            redisTemplate.opsForHash().putAll(key,map);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * hashSet 设置时间
     * @param key
     * @param map
     * @param time
     * @return
     */
    public boolean setHashMap(String key,Map<String,Object> map,long time){
        try {
            redisTemplate.opsForHash().putAll(key,map);
            if(time>0){
                expire(key,time);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 像一张hash表中存入数据，不存在则创建
     * @param key
     * @param item
     * @param value
     * @return
     */
    public boolean setHash(String key,String item,Object value){
        try {
            redisTemplate.opsForHash().put(key,item,value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 在表中放入数据，添加时间
     * @param key
     * @param item
     * @param value
     * @param time
     * @return
     */
    public boolean setHash(String key,String item,Object value,long time){
        try {
            redisTemplate.opsForHash().put(key,item,value);
            if(time>0){
                expire(key,time);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     *删除表中的值
     * @param key
     * @param item
     */
    public void delHash(String key,Object ... item){
        redisTemplate.opsForHash().delete(key,item);
    }

    /**
     * 判断表中是否存在值
     * @param key
     * @param item
     * @return
     */
    public boolean hasHash(String key,String item){
        return redisTemplate.opsForHash().hasKey(key,item);
    }

    /**
     * hash递增，如果不存在，就会新增，并把新增后的值返回
     * @param key
     * @param item
     * @param by
     * @return
     */
    public double hIncr(String key,String item,double by){
        return redisTemplate.opsForHash().increment(key,item,by);
    }

    /**
     * hash递减，
     * @param key
     * @param item
     * @param by
     * @return
     */
    public double hDecr(String key, String item,double by){
        return redisTemplate.opsForHash().increment(key,item,-by);
    }

    //=========================Set===========================================

    /**
     * 获取值
     * @param key
     * @return
     */
    public Set<Object> sGet(String key) {
        try {
            return redisTemplate.opsForSet().members(key);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * 根据value从set中查询，是否存在
     * @param key
     * @param value
     * @return
     */
    public boolean sHasKey(String key,Object value){
        try {
            return redisTemplate.opsForSet().isMember(key,value);
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 将数据放入set缓存中
     * @param key
     * @param values 可以是多个
     * @return 成功的个数
     */
    public long sSet(String key,Object ... values){
        try {
            return redisTemplate.opsForSet().add(key,values);
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }

    /**
     * 存值，添加时间
     * @param key
     * @param time
     * @param values
     * @return
     */
    public long sSetAndTime(String key,long time,Object ... values){
        try {
            Long add = redisTemplate.opsForSet().add(key, values);
            if(time>0){
                expire(key,time);
            }
            return add;
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }

    /**
     * 获取缓存的长度
     * @param key
     * @return
     */
    public long sGetSetSize(String key){
        try {
            return redisTemplate.opsForSet().size(key);
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }

    /**
     * 移除值为values的数据
     * @param key
     * @param values
     * @return
     */
    public long setRemove(String key,Object ... values){
        try {
            return redisTemplate.opsForSet().remove(key,values);
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }


    //===============================List===================================

    /**
     * 获取list的缓存内容
     * @param key
     * @param satrt
     * @param end
     * @return
     */
    public List<Object> lGet(String key,long satrt,long end){
        try {
            return redisTemplate.opsForList().range(key,satrt,end);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }


    /**
     * 获取list的缓存长度
     * @param key
     * @return
     */
    public long lGetListSize(String key){
        try {
            return redisTemplate.opsForList().size(key);
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }

    /**
     * 通过索引获取值
     * @param key
     * @param index index>=0时，0是表头，1是第二个元素，index<0时 -1是表尾，-2是倒数第二个元素
     * @return
     */
    public Object lGetIndex(String key,long index){
        try {
            return redisTemplate.opsForList().index(key,index);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * 将list放入缓存
     * @param key
     * @param values
     * @return
     */
    public boolean lSet(String key,Object values){
        try {
            redisTemplate.opsForList().rightPush(key,values);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 将list放入缓存,设置时间
     * @param key
     * @param values
     * @param time
     * @return
     */
    public boolean lSet(String key,long time,Object values){
        try {
            redisTemplate.opsForList().rightPush(key,values);
            if(time>0){
                expire(key,time);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 将list放入缓存
     * @param key 键
     * @param value 值
     * @return
     */
    public boolean lSet(String key, List<Object> value) {
        try {
            redisTemplate.opsForList().rightPushAll(key, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 将list放入缓存
     * @param key 键
     * @param value 值
     * @param time 时间(秒)
     * @return
     */
    public boolean lSet(String key, List<Object> value, long time) {
        try {
            redisTemplate.opsForList().rightPushAll(key, value);
            if (time > 0) {
                expire(key, time);
            }
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 根据索引修改list中的某条数据
     * @param key 键
     * @param index 索引
     * @param value 值
     * @return
     */
    public boolean lUpdateIndex(String key, long index,Object value) {
        try {
            redisTemplate.opsForList().set(key, index, value);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    /**
     * 移除N个值为value
     * @param key 键
     * @param count 移除多少个
     * @param value 值
     * @return 移除的个数
     */
    public long lRemove(String key,long count,Object value) {
        try {
            Long remove = redisTemplate.opsForList().remove(key, count, value);
            return remove;
        } catch (Exception e) {
            e.printStackTrace();
            return 0;
        }
    }

    //============================ZSet==============================================

    /**
     * 查询集合中指定顺序的值
     * @param key
     * @param start
     * @param end
     * @return 返回一个有序集合，score小的在前面
     */
    public Set<Object> range(String key,long start,long end){
        return redisTemplate.opsForZSet().range(key,start,end);
    }

    /**
     * 查询集合中指定顺序的值
     * @param key
     * @param start
     * @param end
     * @return 返回一个有序集合，score大的在前面
     */
    public Set<Object> revRange(String key,long start,long end){
        return redisTemplate.opsForZSet().reverseRange(key,start,end);
    }

    /**
     * 查询集合中指定顺序的值和score
     * @param key
     * @param start
     * @param end
     * @return
     */
    public Set<ZSetOperations.TypedTuple<Object>> rangeWithScore(String key,int start,int end){
        return redisTemplate.opsForZSet().rangeWithScores(key,start,end);
    }

    /**
     * 根据score的值获取满足条件的集合
     * @param key
     * @param max
     * @param min
     * @return
     */
    public Set<Object> sortRange(String key,int max,int min){
        return redisTemplate.opsForZSet().rangeByScore(key,min,max);
    }

    /**
     * 添加一个元素
     * @param key
     * @param value
     * @param score
     */
    public void add(String key,Object value,double score){
        redisTemplate.opsForZSet().add(key,value,score);
    }

    /**
     * 删除元素
     * @param key
     * @param value
     */
    public void remove(String key,String value){
        redisTemplate.opsForZSet().remove(key,value);
    }

    /**
     * 修改score的值
     * @param key
     * @param value
     * @param score
     * @return
     */
    public Double incrScore(String key,String value,double score){
        return redisTemplate.opsForZSet().incrementScore(key,value,score);
    }

    /**
     * 获取value对应的score的值
     * @param key
     * @param value
     * @return
     */
    public Double score(String key,String value){
        return redisTemplate.opsForZSet().score(key,value);
    }

}
